from django.contrib.auth.mixins import LoginRequiredMixin
from django import http

from utils.response_code import RETCODE


class LoginRequiredJSONMixin(LoginRequiredMixin):
    """自定义判断用户是否登录的扩展类：用来返回JSON数据"""

    # 因为在父类 LoginRequiredMixin中已经实现了验证用户是否登录，在这里我们只需要
    # 重写父类的 handle_no_permission() 方法即可，实现返回JSON数据
    def handle_no_permission(self):
        """直接响应JSON数据"""
        return http.JsonResponse({'code':RETCODE.SESSIONERR, 'errmsg':'用户未登录'})








